3D plots (meshes)

PlantGeom uses Makie.jl to make 3d mesh plots. It also uses MeshViz.jl that is also based on Makie.

This way the plots you make using PlantGeom support all the nice possibilities offered by Makie, such as making sub-plots, interactive plots...

Interactive plot

Here comes the fun part! We can make 3D representations of the plants based on the geometry of each of its nodes.

If you read your MTG from an OPF file, the 3D geometry should already be computed, so you just have to viz() the MTG.

Because we're plotting the interactive plot in the webpage, we must use JSServe first (no need when using Julia from the REPL or VS Code):

using JSServe
Page(exportable=true, offline=true)

Then we can plot our interactive 3D plant:

using PlantGeom, WGLMakie
mtg = read_opf(joinpath(dirname(dirname(pathof(PlantGeom))),"test","files","simple_plant.opf"))
viz(mtg)

Note that the plot is interactive. This is because we use WGLMakie as a plotting backend. You can also use GLMakie for better performance, or CairoMakie if you want a fast, non-interactive plot.

Colors

Note about the backend

In this section, we will use the coffee plant provided as an example OPF file from the package. This one is more realistic than the tiny plant shown above. But because it is bigger, we will only provide static images instead of interactive plots. If you want to plot the interactive plots, you can execute the example code from below using GLMakie or WGLMakie instead of CairoMakie simply by replacing:

using CairoMakie

By:

using GLMakie

Set-up

The first step is to compute the node meshes using the reference meshes and the transformation matrices. This is done very easily by mapping refmesh_to_mesh! to each node of the MTG like so:

using MultiScaleTreeGraph
transform!(mtg, refmesh_to_mesh!)

Default colors

The default behavior of viz(mtg) -without providing colors- is to use the color of each reference mesh as the color of the corresponding node mesh. In other words, a leaf in a tree will be colored with the same color as the reference mesh used to represent it. This reference mesh is available as an attribute in the root node of the MTG. We can extract those reference meshes like so:

using PlantGeom, CairoMakie
CairoMakie.activate!()

file = joinpath(dirname(dirname(pathof(PlantGeom))),"test","files","coffee.opf")
mtg = read_opf(file)

ref_meshes = get_ref_meshes(mtg)

RefMeshes(RefMesh[RefMesh{String, Meshes.SimpleMesh{3, Float64, Vector{Meshes.Point3}, Meshes.SimpleTopology{Meshes.Connectivity}}, Phong{Float64, ColorTypes.RGBA{Float64}}, Vector{Meshes.Point3}, Vector{Meshes.Point2}}("Mesh0", 50 SimpleMesh{3,Float64}, Meshes.Point3[Point(-0.0, -0.9980268, 0.062789984), Point(-0.0, -0.9822871, -0.18738201), Point(-0.0, -0.95105654, 0.30901685), Point(-0.0, -0.90482676, -0.4257799), Point(-0.0, -0.84432787, 0.5358269), Point(-0.0, -0.7705131, -0.6374241), Point(-0.0, -0.68454736, 0.72896844), Point(-0.0, -0.58778507, -0.8090171), Point(-0.0, -0.4817542, 0.87630635), Point(-0.0, -0.36812392, -0.9297768)  …  Point(-0.0, 0.3681251, -0.9297763), Point(-0.0, 0.48175377, 0.8763066), Point(-0.0, 0.58778507, -0.8090171), Point(-0.0, 0.68454695, 0.7289689), Point(-0.0, 0.7705131, -0.6374241), Point(-0.0, 0.84432787, 0.5358269), Point(-0.0, 0.9048272, -0.4257791), Point(-0.0, 0.9510565, 0.30901715), Point(-0.0, 0.9822873, -0.18738106), Point(-0.0, 0.9980268, 0.06279098)], Meshes.Point2[Point(1.0, 3.078761), Point(1.0, 3.33009), Point(1.0, 2.827433), Point(1.0, 3.581416), Point(1.0, 2.576106), Point(1.0, 3.832743), Point(1.0, 2.32478), Point(1.0, 4.084071), Point(1.0, 2.073452), Point(1.0, 4.335399)  …  Point(1.0, 5.089381), Point(1.0, 1.068141), Point(1.0, 5.340707), Point(1.0, 0.816814), Point(1.0, 5.592034), Point(1.0, 0.565487), Point(1.0, 5.843362), Point(1.0, 0.314161), Point(1.0, 6.094688), Point(1.0, 0.062827)], Phong{Float64, ColorTypes.RGBA{Float64}}(RGBA{Float64}(0.0,0.0,0.0,0.0), RGBA{Float64}(0.3,0.45,0.0,0.0), RGBA{Float64}(0.35,0.3,0.0,1.0), RGBA{Float64}(0.2,0.25,0.0,0.0), 10.0), true), RefMesh{String, Meshes.SimpleMesh{3, Float64, Vector{Meshes.Point3}, Meshes.SimpleTopology{Meshes.Connectivity}}, Phong{Float64, ColorTypes.RGBA{Float64}}, Vector{Meshes.Point3}, Nothing}("Mesh1", 12 SimpleMesh{3,Float64}, Meshes.Point3[Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0), Point(0.0, 0.0, 1.0)], nothing, Phong{Float64, ColorTypes.RGBA{Float64}}(RGBA{Float64}(0.0,0.0,0.0,0.0), RGBA{Float64}(0.1,0.8,0.1,0.5), RGBA{Float64}(0.1,0.7,0.1,1.0), RGBA{Float64}(0.1,0.9,0.2,0.5), 10.0), true)])

Then we can plot them in sequence:

viz(ref_meshes)

Here we are looking at the reference meshes used to build the plant. Those meshes are then transformed by transformation matrices from each node to make the mesh of that node. So by default the color used for the nodes will be taken from these reference meshes.

If we plot the coffee plant without providing any color, we would get:

viz(mtg)

Single color

Now we can change the color of all meshes by providing a single color:

viz(mtg, color = :gray87)

Map color to reference meshes

We can also associate a new color to each reference mesh.

We can get the default color of each reference mesh by using:

get_ref_meshes_color(ref_meshes)

Now we know the first reference mesh is the cylinder (it is brown) and the second one is the leaf (it is green).

To update their colors we can simply pass the new colors as a dictionary mapping colors to reference meshes like so:

viz(mtg, color = Dict(1 => :gray87, 2 => "#42A25ABD"))

If we want to update the second reference mesh only (the leaves), we would do:

viz(mtg, color = Dict(2 => "#42A25ABD"))

Map color to attributes

Maybe the most interesting coloring option is to color by attribute.

Indeed, each node in the MTG can have specific attributes, e.g. an area, a temperature...

You can see which attributes are available in an MTG using:

print(names(mtg))

[:FileName, :Variety, :ref_meshes, :Plot, :Treatment, :File, :Name, :_cache_d9b4f7f3c3467a55ad26f362065777c471aee4c7, :geometry, :XEuler, :Length, :Width, :Stifness, :YInsertionAngle, :Plagiotropy, :Phyllotaxy, :StiffnessAngle, :Area, :XInsertionAngle]

We can see that we have an attribute called :Area. Let's color each organ by its area:

viz(mtg, color = :Area)

Of course all Makie commands are available. For example we can zoom-in the plot using scale!, and add a colorbar:

f, ax, p = viz(mtg, color = :Area)
CairoMakie.scale!(p, 1.5, 1.5, 1.5) # we zoom-in a little bit
CairoMakie.Colorbar(f[1,2], label = "Area")
f

We can see that the colorbar is only in relative values (0-1). If you need absolute values, you can use PlantGeom's colorbar instead:

f, ax, p = viz(mtg, color = :Area)
colorbar(f[1, 2], p)
f

Map color to vertices

# Compute the z position of each vertices in each mesh:
transform!(mtg, :geometry => (x -> [i.coords[3] for i in x.mesh.vertices]) => :z, ignore_nothing = true)
viz(mtg, color = :z, showfacets = true, color_vertex = true)

Map time step to color

The MTG attributes can have several values, for example a value for each time step of a simulation. For example, let's make a dummy variable with 12 time-steps, each value being the area time the time step:

transform!(mtg, :Area => (x -> [x*i for i in 1:12]) => :dummy_var, ignore_nothing = true)

Now we can plot the plant with the color of each organ being the value of the dummy variable at time step 1 using the index keyword argument:

f, ax, p = viz(mtg, color = :dummy_var, index = 1)
colorbar(f[1, 2], p)
f

We can even make a video out of it:

f, ax, p = viz(mtg, color = :dummy_var, index = 1)
colorbar(f[1, 2], p)

record(f, "coffee_steps.mp4", 1:12, framerate=2) do timestep
    p.index[] = timestep
end

"coffee_steps.mp4"